package gulimall.config;

import com.alibaba.cloud.nacos.NacosDiscoveryProperties;
import com.alibaba.cloud.nacos.loadbalancer.NacosLoadBalancer;
import org.springframework.cloud.client.ServiceInstance;
import org.springframework.cloud.client.loadbalancer.reactive.ReactiveLoadBalancer;
import org.springframework.cloud.loadbalancer.annotation.LoadBalancerClient;
import org.springframework.cloud.loadbalancer.annotation.LoadBalancerClients;
import org.springframework.cloud.loadbalancer.core.ServiceInstanceListSupplier;
import org.springframework.cloud.loadbalancer.support.LoadBalancerClientFactory;
import org.springframework.context.annotation.Bean;
import org.springframework.core.env.Environment;

import javax.annotation.Resource;

/**
 * @author HLF
 * @date 2022/8/22 20:09
 */
@LoadBalancerClients(value = {
        @LoadBalancerClient(value = "gulimall-product")
}, defaultConfiguration = LoadBalancerConfig.class)
public class LoadBalancerConfig {

    @Resource
    private NacosDiscoveryProperties nacosDiscoveryProperties;

    /**
     * 基于Nacos注册中心权重的负载均衡配置
     */
    @Bean
    public ReactiveLoadBalancer<ServiceInstance> nacosLoadBalancer(
            LoadBalancerClientFactory loadBalancerClientFactory,
            Environment environment) {

        String name = environment.getProperty(LoadBalancerClientFactory.PROPERTY_NAME);
        return new NacosLoadBalancer(
                loadBalancerClientFactory.getLazyProvider(name, ServiceInstanceListSupplier.class),
                name, nacosDiscoveryProperties
        );
    }
}
